# 人物检测
import cv2
import argparse

from ultralytics import YOLO


def human_detect(pt,img_in):
    # 加载 YOLO 模型
    model = YOLO(pt);  # 使用 YOLOv8 模型
    # 读取图片
    img = cv2.imread(img_in);

    # 检查图片是否加载成功
    if img is None:
        print("图片加载失败，请检查路径！")
        exit()

    # 转换颜色格式（YOLO 使用 RGB 格式）
    img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB);
    # 检测
    results = model.predict(source=img_rgb, save=False,verbose=False) ;
    # 判断是否检测出人
    if len(results[0].boxes) == 0:
        print("未检测出人脸")
        exit()
        
    else:
        print( "检测完成！") 
if __name__=="__main__":
    parser=argparse.ArgumentParser(description='human_detect')
    parser.add_argument("--address")
    parser.add_argument("--ptaddress")
    args=parser.parse_args()
    human_detect(args.ptaddress,args.address)
    
